package com.hy;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * Created With IntelliJ IDEA.
 * Descriptions:串联所有单词的子串
 * 给定一个字符串 s 和一个字符串数组 words。 words 中所有字符串 长度相同。
 *  s 中的 串联子串 是指一个包含  words 中所有字符串以任意顺序排列连接起来的子串。
 *
 * 例如，如果 words = ["ab","cd","ef"]， 那么 "abcdef"， "abefcd"，"cdabef"， "cdefab"，"efabcd"， 和 "efcdab"
 * 都是串联子串。 "acdbef" 不是串联子串，因为他不是任何 words 排列的连接。
 * 返回所有串联子串在 s 中的开始索引。你可以以 任意顺序 返回答案。
 *
 * 示例 1：
 * 输入：s = "barfoothefoobarman", words = ["foo","bar"]
 * 输出：[0,9]
 * 解释：因为 words.length == 2 同时 words[i].length == 3，连接的子字符串的长度必须为 6。
 * 子串 "barfoo" 开始位置是 0。它是 words 中以 ["bar","foo"] 顺序排列的连接。
 * 子串 "foobar" 开始位置是 9。它是 words 中以 ["foo","bar"] 顺序排列的连接。
 * 输出顺序无关紧要。返回 [9,0] 也是可以的。
 *
 * 示例 2：
 *
 * 输入：s = "wordgoodgoodgoodbestword", words = ["word","good","best","word"]
 * 输出：[]
 * 解释：因为 words.length == 4 并且 words[i].length == 4，所以串联子串的长度必须为 16。
 * s 中没有子串长度为 16 并且等于 words 的任何顺序排列的连接。
 * 所以我们返回一个空数组。
 *
 * User:Mr.Du
 * Date:2024/1/11
 * Time:15:50
 */
public class FindSubstring {

    /**
     * 朴素哈希
     * @param s
     * @param words
     * @return
     */
    public List<Integer> findSubstring(String s, String[] words) {
        int m = s.length(), n = words.length, w = words[0].length();
        List<Integer> res = new ArrayList<>();
        // 统计words中字符出现次数
        Map<String,Integer> mp = new HashMap<>();
        for(String ss : words) mp.put(ss, mp.getOrDefault(ss, 0) + 1);
        for(int i = 0;i + n * w <= m;i++){
            String sub = s.substring(i, i + n * w);
            // 统计sub子串中长度为w的子串出现次数
            Map<String,Integer> cur = new HashMap<>();
            for(int j = 0;j < sub.length();j += w){
                String ss = sub.substring(j, j + w);
                // ss不在mp中，直接退出
                if(!mp.containsKey(ss)) break;
                cur.put(ss, cur.getOrDefault(ss, 0) + 1);
            }
            if(cur.equals(mp)) res.add(i);
        }
        return res;
    }
}
